package com.zjc.agent.ai.utils;


import lombok.extern.slf4j.Slf4j;

/**
 * ThreadLocal工具类，用于存储和获取请求token
 * @author zjc
 */
@Slf4j
public class TokenContextHolder {
    private static final ThreadLocal<String> TOKEN_HOLDER = new ThreadLocal<>();

    /**
     * 设置token
     * @param token 请求token
     */
    public static void setToken(String token) {
        TOKEN_HOLDER.set(token);
    }

    /**
     * 获取token
     * @return 当前token
     */
    public static String getToken() {
        return TOKEN_HOLDER.get();
    }

    /**
     * 清除token
     */
    public static void clear() {
        TOKEN_HOLDER.remove();
    }
}
